package com.chtw.oasystem.interceptor;

import com.alibaba.fastjson.JSONObject;
import com.chtw.oasystem.common.response.RespUtils;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.PrintWriter;

public class LoginInterceptor implements HandlerInterceptor {
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        Object user = request.getSession().getAttribute("user_name");
        response.setContentType("application/json;charset=utf-8");
        if (user == null) {
            //未登陆，返回登陆页面
            responseMessage(response, response.getWriter(), RespUtils.NotLoginFail());
            return false;
        } else {
            //已登陆，放行请求
            return true;
        }

    }

    @Override
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {

    }

    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {

    }

    private void responseMessage(HttpServletResponse response, PrintWriter out, RespUtils apiResponse) {
        response.setContentType("application/json;charset=utf-8");
        out.print(JSONObject.toJSONString(apiResponse));
        out.flush();
        out.close();
    }
}
